from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
import numpy

numpy_inc = '-I' + numpy.__path__[0] +  '/core/include/'

ext_modules=[
  Extension("bwshrink.mrf_wavelet_icm",
    ["bwshrink/mrf_wavelet_icm.pyx"],
    extra_compile_args=[numpy_inc]
    ),

  Extension("bwshrink.logistic",
    ["bwshrink/logistic.pyx"],
    extra_compile_args=[numpy_inc]
    ),
#
#  Extension("bwshrink.mrf_wavelet_mcmc",
#    ["bwshrink/mrf_wavelet_mcmc.pyx"],
#    extra_compile_args=[numpy_inc]
#    ),
#
#  Extension("bwshrink.biweight",
#    ["bwshrink/biweight.pyx"],
#    extra_compile_args=[numpy_inc]
#    ),
  ]


setup(
  name='bayes-wavelet-shrinkage',
  version='0.2',
  description='Bayesian Wavelet Shrinkage using MRF priors',
  author='Harry Zuzan',
  author_email='iliketobicycle@gmail.com',
  cmdclass={'build_ext':build_ext},
  url='http://code.google.com/p/bayes-wavelet-shrinkage-mrf',
  packages=['bwshrink'],
  package_dir={'bwshrink':'bwshrink',},
  ext_modules=ext_modules,
#  scripts=['scripts/describe_gpu'],
  )

